package com.example.file_demo.util;

import java.io.*;
import java.nio.charset.Charset;
import java.util.zip.*;

public class ZipUtils {

    /**
     * 解压ZIP文件到指定路径
     *
     * @param fileInputStream fileInputStream流
     * @param destDir     解压目标目录
     * @throws IOException 如果发生I/O错误
     */
    public static void unzip(FileInputStream fileInputStream, String destDir) {
        File destDirectory = new File(destDir);
        if (!destDirectory.exists()) {
            destDirectory.mkdirs();
        }
        try (ZipInputStream zipIn = new ZipInputStream(fileInputStream, Charset.forName("GBK"))) {
            ZipEntry entry = zipIn.getNextEntry();
            // 遍历ZIP文件中的每个条目
            while (entry != null) {
                String filePath = destDir + File.separator + entry.getName();
                if (!entry.isDirectory()) {
                    // 如果条目是文件，则解压文件
                    extractFile(zipIn, filePath);
                } else {
                    // 如果条目是目录，则创建目录
                    File dir = new File(filePath);
                    dir.mkdirs();
                }
                zipIn.closeEntry();
                entry = zipIn.getNextEntry();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 解压单个文件
     *
     * @param zipIn    ZipInputStream对象
     * @param filePath 文件解压后的路径
     * @throws IOException 如果发生I/O错误
     */
    private static void extractFile(ZipInputStream zipIn, String filePath) throws IOException {
        try (BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(filePath))) {
            byte[] bytesIn = new byte[4096];
            int read;
            while ((read = zipIn.read(bytesIn)) != -1) {
                bos.write(bytesIn, 0, read);
            }
        }
    }
}